from ultralytics import YOLO


def yolo(source):

    model = YOLO("res/best.pt", task="detect")
    result = model.predict(source=source)
    result[0].plot()
    return result
